import time

from taskinfo.UieText import UieText

if __name__ == '__main__':
    schema = "车牌号码,船名,净重,过磅时间".split(',')
    ie = UieText(schema_list=schema, onnx_home=r'D:\ocr\train_data\information_extraction\tianjingang_text')
    start_time = time.perf_counter()
    result = ie([
        "5:31系;l  25%;任务;电子运单;天津港远航码头电子运单;车牌号码：津A08528F;挂车号码：津C7379挂;计划编号：QC-20240502-02;货类:澳洲精粉矿;船名:诚信16;船次:YT017;场地:前沿货场;货位:A2-4,C7-1;回皮磅房：前沿1号磅，前沿2号磅;过重磅房：前3号磅，前沿4号磅，前沿5号磅;货主：荣钢;毛重：48680;代理：荣程中联;皮重：19360;流向:荣钢 (天津);净重：29320;车队：荣程智运(氢能源);运费:18;派车时间：2024-05-0222:52:38;过磅时间：2024-05-0305：23：46;合同编号：;口"])
    end_time = time.perf_counter()
    print("time:", end_time - start_time)
    print(result)
    for key, value in result[0].items():
        print(key, value)
